package cn.ztuo.bitrade.config;

import cn.ztuo.bitrade.entity.ContractCoin;
import cn.ztuo.bitrade.entity.ContractPriceControl;
import cn.ztuo.bitrade.ext.OrdinalToEnumConverterFactory;
import cn.ztuo.bitrade.interceptor.LogInterceptor;
import cn.ztuo.bitrade.interceptor.OutExcelInterceptor;
import cn.ztuo.bitrade.interceptor.SessionInterceptor;
import cn.ztuo.bitrade.service.ContractCoinService;
import cn.ztuo.bitrade.service.ContractPriceControlService;
import cn.ztuo.bitrade.vo.contractPriceControlVo.ContractPriceControlFindReqVo;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.web.servlet.FilterRegistrationBean;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.support.ResourceBundleMessageSource;
import org.springframework.format.FormatterRegistry;
import org.springframework.web.cors.CorsConfiguration;
import org.springframework.web.cors.UrlBasedCorsConfigurationSource;
import org.springframework.web.filter.CorsFilter;
import org.springframework.web.servlet.config.annotation.InterceptorRegistry;
import org.springframework.web.servlet.config.annotation.ResourceHandlerRegistry;
import org.springframework.web.servlet.config.annotation.WebMvcConfigurerAdapter;

import java.util.*;

/**
 * @author Administrator
 */
@Configuration
public class ApplicationConfig extends WebMvcConfigurerAdapter {
    @Autowired
    private ContractPriceControlService priceControlService;
    /**
     * 国际化
     *
     * @return
     */
    @Bean(name = "messageSource")
    public ResourceBundleMessageSource getMessageSource() {
        ResourceBundleMessageSource resourceBundleMessageSource = new ResourceBundleMessageSource();
        resourceBundleMessageSource.setDefaultEncoding("UTF-8");
        resourceBundleMessageSource.setBasenames("i18n/messages", "i18n/ValidationMessages");
        resourceBundleMessageSource.setCacheSeconds(3600);
        return resourceBundleMessageSource;
    }

    @Bean
    public FilterRegistrationBean corsFilter() {
        UrlBasedCorsConfigurationSource source = new UrlBasedCorsConfigurationSource();
        CorsConfiguration config = new CorsConfiguration();
        config.addAllowedOrigin("*");
        config.setAllowCredentials(true);
        config.addAllowedHeader("*");
        config.addAllowedMethod("*");
        source.registerCorsConfiguration("/**", config);
        FilterRegistrationBean bean = new FilterRegistrationBean(new CorsFilter(source));
        bean.setOrder(0);
        return bean;
    }
    @Override
    public void addResourceHandlers(ResourceHandlerRegistry registry) {
        registry.addResourceHandler("/asset/**").addResourceLocations("classpath:/asset/");
        super.addResourceHandlers(registry);
    }

    @Override
    public void addFormatters(FormatterRegistry registry) {
        registry.addConverterFactory(new OrdinalToEnumConverterFactory());
        super.addFormatters(registry);
    }
    @Override
    public void addInterceptors(InterceptorRegistry registry) {
        registry.addInterceptor(new SessionInterceptor()).addPathPatterns("/**")
                .excludePathPatterns("/code/sms-provider/**","/captcha","/system/employee/sign/in",
                        "/system/employee/check","/system/employee/logout","/finance/manually/**","/swagger-ui.html/**/","/contract/contract-order/statistics","/contract/contract-order/updateContractDetailTypeAndUnit");
        registry.addInterceptor(new LogInterceptor()).addPathPatterns("/**");
        registry.addInterceptor(new OutExcelInterceptor()).addPathPatterns("/**/out-excel");
        super.addInterceptors(registry);
    }

    /**
     * 配置交易处理类
     * @param contractCoinService
     * @return
     */
    @Bean(name="contractPriceControlSymbols")
    public Map<String, TreeMap<Date, ContractPriceControl>> getContractPriceControlSymbols(ContractCoinService contractCoinService, ContractPriceControlService priceControlService){
        Map<String, TreeMap<Date, ContractPriceControl>> map = new HashMap<>();
        List<ContractCoin> coins = contractCoinService.findAllEnabled();
        for(ContractCoin coin:coins) {
            TreeMap<Date, ContractPriceControl> treeMap  = new TreeMap<>(Comparator.naturalOrder());
            ContractPriceControlFindReqVo reqVo = new ContractPriceControlFindReqVo();
            reqVo.setEndTime(new Date());
            reqVo.setStartTime(new Date());
            reqVo.setStatus(1);
            reqVo.setType(1);
            reqVo.setSymbol(coin.getSymbol());
            List<ContractPriceControl> controls = priceControlService.findAllCanable(reqVo);
            for (ContractPriceControl priceControl:controls){
                if(priceControl.getEndTime().after(new Date())){
                    synchronized (treeMap) {
                        treeMap.put(priceControl.getStartTime(),priceControl);
                    }
                }else{
                    priceControl.setStatus(3);
                    priceControlService.saveAndFlush(priceControl);
                }
            }
            map.put(coin.getSymbol(),treeMap);
        }
        return map;
    }

}
